3272e413a0af438406aba74bb3fa4e3d10a4f515,h2o-algos/src/test/java/hex/word2vec/Word2VecTest.java,Word2VecTest,testW2V_SG_NS,#,122

Before Change


      job = new Word2Vec(parms);
      job.train();
      w2vm = job.get();
      printResults(w2vm.findSynonyms("dog",5));
    } finally {
      if( fr  != null ) fr .remove();
      if( job != null) job.remove();

After Change


      job = new Word2Vec(parms);
      job.train();
      w2vm = job.get();
      HashMap hm = w2vm.findSynonyms("dog",10);
      printResults(hm);
      Assert.assertTrue(hm.containsKey("dogs"));
    } finally {
      if( fr  != null ) fr .remove();